torch
transformers
scikit-learn
stable-baselines3
dacite
gymnasium[atari,accept-rom-license,mujoco]
